mamba install scikit-learn accelerate sentencepiece jupyter nvitop conda-forge::transformers pytorch torchvision torchaudio pytorch-cuda=12.1 matplotlib -c pytorch -c nvidia -y
